/**
 * Copyright 2023 Huawei Technologies Co., Ltd
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include "src/extendrt/kernel/kernel_selector/kernel_selector.h"
#include "src/extendrt/kernel/kernel_selector/nnacl_first_kernel_selector.h"

namespace mindspore::kernel {
std::vector<const KernelLib *> KernelSelector::Candidates(const PrimitiveType &op_type, const KernelAttr &require,
                                                          const std::string &backend, Format format) {
  std::vector<const KernelLib *> results;
  for (const auto &iter : KernelLibRegister::Instance().GetAllLibs()) {
    const auto *kernel_lib = iter.second;
    if (kernel_lib->Support(op_type, require, backend, format)) {
      results.emplace_back(kernel_lib);
    }
  }
  return results;
}

std::shared_ptr<KernelSelector> CreateKernelSelector(const std::shared_ptr<lite::CompileOption> &compile_option) {
  return std::make_shared<NNACLFirstKernelSelector>(compile_option);
}
}  // namespace mindspore::kernel
